short python program